a2633871e6a47a8a22479f8f41082f6662f0c76a,core/generator/test/jetbrains/mps/generator/impl/plan/CheckpointModelTest.java,CheckpointModelTest,createModelWithOneCheckpoint,#,86

Before Change


    myErrors.checkThat("Generation succeeds", genStatus.isOk(), CoreMatchers.equalTo(true));
    CrossModelEnvironment cme = new CrossModelEnvironment(tmProvider);
    // XXX shall it be CME to give access to module with checkpoint models? Is there better way to find out cpModel?
    myErrors.checkThat("CrossModelEnvironment.hasState", cme.hasState(mr, planIdentity), CoreMatchers.equalTo(true));

    SModule checkpointModule = tmProvider.getCheckpointsModule();
    final SModelName cpModelName = CrossModelEnvironment.createCheckpointModelName(m.getReference(), cp1);

After Change


    GenerationFacade genFacade = new GenerationFacade(mpsProject.getRepository(), opt).transients(tmProvider);
    GenerationStatus genStatus = genFacade.process(new EmptyProgressMonitor(), m);
    myErrors.checkThat("Generation succeeds", genStatus.isOk(), CoreMatchers.equalTo(true));
    CrossModelEnvironment cme = new CrossModelEnvironment(tmProvider, new ModelStreamProviderImpl());
    // XXX shall it be CME to give access to module with checkpoint models? Is there better way to find out cpModel?

    SModule checkpointModule = tmProvider.getCheckpointsModule();
    final SModelName cpModelName = CrossModelEnvironment.createCheckpointModelName(m.getReference(), cp1);
    SModel cpModel = null;
    for (SModel trm : checkpointModule.getModels()) {
      if (cpModelName.equals(trm.getName())) {
        cpModel = trm;
        break;
      }
    }
    myErrors.checkThat("Checkpoint model", cpModel, CoreMatchers.notNullValue());
    ModelCheckpoints modelCheckpoints = cme.getState(m, planIdentity);
    myErrors.checkThat("CrossModelEnvironment: state present", modelCheckpoints, CoreMatchers.notNullValue());
    CheckpointState cpState = modelCheckpoints.find(cp1);
    myErrors.checkThat("CheckpointState present", cpState, CoreMatchers.notNullValue());
    if (cpState != null) {